import torch

ckpt = torch.load('../checkpoint/distmult.ckpt')
ents=ckpt["ent_embeddings.weight"]
for ent in ents:
    print(ent)

